40. 组合总和 II

发表于 | 分类于 | 阅读次数：

链接：https://leetcode-cn.com/problems/combination-sum-ii/

题目描述

给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明：

所有数字（包括目标数）都是正整数。
解集不能包含重复的组合。
示例 1:

1
2
3
4
5
6
7
8
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

1
2
3
4
5
6
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
  [1,2,2],
  [5]
]

分析

这道题也是非常标准的回溯算法的步骤，唯一需要注意的两个点是：

  1. 求总和问题一般会先排个序，然后顺序相加，当result>target时，后面还没相加的所有的因为排序，就都剪枝了。
  2. 组合问题保证不重复的方法，可以想象一颗递归树，每次不能长出来两个相同的枝，否则就会有重复。

综合，本题答案参考如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class Solution(object):
    def __init__(self):
        self.result_all = None
    
    def combinationSum2(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        self.result_all = []
        candidates = sorted(candidates)
        self.dfs(candidates, target, 0, [])
        return self.result_all
    
    def dfs(self, candidates, target, start, result):
        if sum(result) == target:
            self.result_all.append(result[:])
            return
        pre_nums = []
        for i in range(start, len(candidates)):
            num = candidates[i]
            if num in pre_nums:
                continue
            if sum(result) + num > target:
                break
            pre_nums.append(num)
            result.append(num)
            self.dfs(candidates, target, i + 1, result)
            result.pop()
        return